from flask import Flask, session, g
# 导入数据库相关包
from exts import db,mail  # 导入创建的db对象
import config  # 导入数据库配置语句
from models import UserModel  # 导入ORM模型建表语句
from flask_migrate import Migrate  # 使用ORM映射语句

# 导入蓝图：blueprints.文件名
# 因为每个py文件中都是bp对象名，为了防止重复，后面加上：as 新名字，就可以在本文件中分别调用
from blueprints.qa import bp as qa_bp
from blueprints.auth import bp as auth_bp

app = Flask(__name__)

app.config.from_object(config)  # 绑定config配置文件，获取到配置
db.init_app(app)  # 绑定db
mail.init_app(app)
migrate = Migrate(app, db)  # ORM模型映射
# 绑定蓝图
app.register_blueprint(qa_bp)
app.register_blueprint(auth_bp)


# 视图函数都在其他文件，此处不需要了
# @app.route('/')
# def hello_world():
#     return 'Hello World!'

@app.before_request
def my_before_request():
    # flask自动解密
    user_id = session.get("user_id")
    # 若未登录，则无cookie，就没有session
    if user_id:
        user = UserModel.query.get(user_id)
        # 拿到user后绑定到全局对象，引入global包”g“
        # 先给g设置一个属性"user"，其属性值为user
        setattr(g,"user", user)
    else:
        # 为防止报错（若没拿到user_id，就不会创建g对象，若后面调用g对象，就会报错）
        # 因此此处创建g对象并设置属性为None
        setattr(g,"user",None)

# 上下文处理器：页面渲染
# 假如用户访问站点的时候，站点的所有页面上都要能够显示这个用户自己的IP地址。
# 每个视图函数中都从请求（request）中获取到访问用户的IP地址，然后呈现到页面中太过麻烦。
# 将处理过程自定义一个上下文处理器，就能够在每个页面中使用
@app.context_processor
def my_context_processor():
    # 返回user的信息
    return {"user":g.user}


if __name__ == '__main__':
    app.run()
